package com.ophiux.utils.common;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.DocumentHelper;
import org.dom4j.Element;
import org.dom4j.Namespace;
import org.dom4j.QName;

import lombok.extern.slf4j.Slf4j;

/**
 * 说明：xml转map工具类<br>
 * <h1>创 建 人: hehailong</h1>
 * 创建日期: 2019年9月25日 下午2:22:52<br>
 * 修改记录：
 */
@Slf4j
public class XmlToMapUtil {
	
	/**
	 * 说明：xml转map<br>
	 * 创 建 人: <b>hehailong</b>
	 * @param xmlStr
	 * @return
	 */
	public static Map<String, Object> parseXmlToMap(String xmlStr) {
		Document doc = parseToDocument(xmlStr);
		Map<String, Object> map = (Map<String, Object>) xml2map(doc.getRootElement());
		return map;
	}

	public static Document parseToDocument(String xmStr) {
		org.dom4j.Document doc = null;
		try {
			return doc = DocumentHelper.parseText(xmStr);

		} catch (DocumentException e) {
			e.printStackTrace();
			log.error("xml生成错误 字符串：{}", xmStr);
		}
		return doc;
	}

	public static Object xml2map(Element element) {

		Map<String, Object> map = new HashMap<String, Object>();
		List<Element> elements = element.elements();

		if (elements.size() == 0) { // 如果没有子节点
			map.put(element.getName(), element.getText());
			if (!element.isRootElement()) {
				return element.getText();
			}
		} else if (elements.size() == 1) { // 只有一个子节点
			map.put(elements.get(0).getName(), xml2map(elements.get(0)));
		} else if (elements.size() > 1) { // 含多个子节点

			Map<String, Element> tempMap = new HashMap<String, Element>();
			for (Element ele : elements) {
				tempMap.put(ele.getName(), ele);
			}
			Set<String> keySet = tempMap.keySet();
			for (String string : keySet) {
				Namespace namespace = tempMap.get(string).getNamespace();
				List<Element> elements2 = element.elements(new QName(string, namespace));
				// 如果同名的数目大于1则表示要构建list
				if (elements2.size() > 1) {
					List<Object> list = new ArrayList<Object>();
					for (Element ele : elements2) {
						list.add(xml2map(ele));
					}
					map.put(string, list);
				} else {
					// 同名的数量不大于1则直接递归去
					map.put(string, xml2map(elements2.get(0)));
				}
			}
		}

		return map;
	}
}
